from read_mnist import *

imgs = read_mnist_images('../datasets/train-images.idx3-ubyte')
labels = read_mnist_labels('../datasets/train-labels.idx1-ubyte')

assert len(imgs)==60000, "Length of imgs is not 60000"
assert imgs.shape[1:]==(28,28), "Image shape is not 28x28"
assert len(labels)==60000, "Length of labels is not 60000"
